import argparse

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('--epoch',type=int,default=25)
    parser.add_argument('--lr', type=float, default=0.001)
    parser.add_argument('--batch_size', type=int, default=1024)
    parser.add_argument('--laten_dim', type=int, default=8,help="FM特征隐向量维度")

    return parser.parse_args()

args = parse_args()